"""
@author: liujun
@file: test.py
@time: 2022/11/24 17:53
"""
import os
import os.path


def tree(path, depth=0):
    if depth == 0:
        print(path)
    items = os.listdir(path)
    for item in items:
        # 输出文件夹中的文件和子文件夹名
        print('|    ' * depth, end='')
        print('|----', item)
        item = os.path.join(path, item)
        if os.path.isdir(item):
            # 递归遍历子目录
            tree(item, depth + 1)
tree(r'D:/STD/PyCharm/PyCharm 2020.3/上课代码/2022-11-24-模块和包')

N=input("请输入你需要的邀请码数量：")
import liujun
liujun.random_num(N)